package com.guocheng.base;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
import com.guocheng.utils.LetterUtils;
import com.guocheng.utils.WrapperUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;

import java.io.Serializable;
import java.util.Collection;
import java.util.List;
import java.util.Map;

/**
 * Service继承该类
 * 包含基本的增删改查
 * <p>
 * 分页查询
 */
public class BaseService<M extends BaseMapper<T>, T> {

    @Autowired
    protected M baseDao;

    /**
     * 添加一条数据
     *
     * @param entity 要添加的数据
     * @return 成功
     */
    @Transactional(readOnly = false)
    public boolean insert(T entity) {
        return SqlHelper.retBool(baseDao.insert(entity));
    }

    /**
     * 根据id删除
     *
     * @param id id
     * @return 成功
     */
    @Transactional(readOnly = false)
    public boolean deleteById(Serializable id) {
        return SqlHelper.retBool(baseDao.deleteById(id));
    }

    /**
     * 根据多id删除
     *
     * @param ids 删除的id列表
     * @return 删除条数
     */
    @Transactional(readOnly = false)
    public int deleteByIds(Collection<? extends Serializable> ids) {
        return baseDao.deleteBatchIds(ids);
    }

    /**
     * 根据封装的条件删除
     *
     * @param entity
     * @return
     */
    @Transactional(readOnly = false)
    public int deleteByEntity(T entity) {
        return baseDao.delete(WrapperUtils.modelToQueryWrapper(entity));
    }

    /**
     * 根据id修改
     *
     * @param entity 修改的实体类（必须携带id）
     * @return 是否成功
     */
    @Transactional(readOnly = false)
    public boolean updateById(T entity) {
        return SqlHelper.retBool(baseDao.updateById(entity));
    }

    /**
     * 根据条件修改
     *
     * @param wareep 修改的条件
     * @param entity 要修改的值
     * @return 修改条数
     */
    @Transactional(readOnly = false)
    public int updateByEntity(T wareep, T entity) {
        return baseDao.update(entity, WrapperUtils.modelToUpdateWrapper(wareep));
    }

    /**
     * 根据条件修改
     *
     * @param entity 要修改的值
     * @param wrapper 修改的条件
     * @return 修改条数
     */
    @Transactional(readOnly = false)
    public int updateByEntity(T entity, UpdateWrapper<T> wrapper) {
        return baseDao.update(entity, wrapper);
    }


    /**
     * 根据id查询
     *
     * @param id id
     * @return 实体类
     */
    public T selectById(Serializable id) {
        return baseDao.selectById(id);
    }

    /**
     * 根据id查询
     *
     * @param ids id
     * @return 实体类
     */
    public List<T> selectByIds(Collection<? extends Serializable> ids) {
        return baseDao.selectBatchIds(ids);
    }

    /**
     * 自定义封装条件查询第一个
     *
     * @param queryWrapper 构造器
     * @return 查询后的实体类
     */
    public T selectOne(Wrapper<T> queryWrapper) {
        return baseDao.selectOne(queryWrapper);
    }

    /**
     * 查询列表
     * @return
     */
    public List<T> selectList(){
        return baseDao.selectList(null);
    }

    /**
     * 自定义封装条件查询
     *
     * @param queryWrapper 构造器
     * @return 查询后的列表
     */
    public List<T> selectList(Wrapper<T> queryWrapper) {
        return baseDao.selectList(queryWrapper);
    }

    /**
     * 根据实体类查询列表
     * @param entity 实体类
     * @return 列表
     */
    public List<T> selectList(T entity){
        QueryWrapper<T> queryWrapper = WrapperUtils.modelToQueryWrapper(entity);
        return baseDao.selectList(queryWrapper);
    }

    /**
     * 根据实体类查询列表
     * @param entity 实体类
     * @return 列表
     */
    public List<T> selectList(T entity,QueryWrapper<T> queryWrapper){
        WrapperUtils.queryWrapperAdd(queryWrapper,entity);
        return baseDao.selectList(queryWrapper);
    }

    /**
     * 根据实体类不分页查询排序
     * @param baseModel 排序内容
     * @return 列表
     */
    public List<T> selectList(BaseModel baseModel){
        QueryWrapper<T> queryWrapper = WrapperUtils.queryWrapperAdd(new QueryWrapper<T>(), baseModel);
        queryWrapper.orderBy(true, baseModel.getIsSort(), LetterUtils.valTocamel(baseModel.getOrderBy(), "_"));
        return baseDao.selectList(queryWrapper);
    }

    /**
     * 根据map查询
     *
     * @param map key:String 查询的字段 value:Objcet 查询的值(无模糊查询)
     * @return 查询列表
     */
    public List<T> selectByMap(Map<String, Object> map) {
        return baseDao.selectByMap(map);
    }

    /**
     * 根据map查询
     *
     * @param map key:String 查询的字段 value:Objcet 查询的值(有模糊查询)
     * @return 查询列表
     */
    public List<T> selectLikeByMap(Map<String, Object> map) {
        QueryWrapper<T> queryWrapper = WrapperUtils.mapToQueryWrapper(new QueryWrapper<T>(), map);
        return baseDao.selectList(queryWrapper);
    }

    /**
     * 分页查询
     *
     * @param page  当前页
     * @param limit 每页个数
     * @return 查询后的数据
     */
    public IPage<T> selectPage(int page, int limit) {
        return selectPage(new QueryWrapper<T>(), page, limit);
    }

    /**
     * 分页查询
     *
     * @param queryWrapper 封装的条件
     * @param page         当前页
     * @param limit        每页个数
     * @return 查询后的数据
     */
    public IPage<T> selectPage(QueryWrapper<T> queryWrapper, int page, int limit) {
        return selectPage(queryWrapper, page, limit, null);
    }

    /**
     * 分页查询
     *
     * @param queryWrapper 封装的条件
     * @param page         当前页
     * @param limit        每页个数
     * @param orderBy      排序字段（升序）
     * @return 查询后的数据
     */
    public IPage<T> selectPage(QueryWrapper<T> queryWrapper, int page, int limit, String orderBy) {
        return selectPage(queryWrapper, page, limit, orderBy, true);
    }

    /**
     * 分页查询
     *
     * @param page         当前页
     * @param limit        每页个数
     * @param orderBy      排序字段（升序）
     * @return 查询后的数据
     */
    public IPage<T> selectPage(int page, int limit, String orderBy) {
        return selectPage(new QueryWrapper<T>(), page, limit, orderBy, true);
    }

    /**
     * 分页查询
     *
     * @param queryWrapper 封装的条件
     * @param page         当前页
     * @param limit        每页个数
     * @param orderBy      排序字段
     * @param sort         升序true 降序false
     * @return 查询后的数据
     */
    public IPage<T> selectPage(QueryWrapper<T> queryWrapper, int page, int limit, String orderBy, boolean sort) {
        IPage<T> userPage = new Page<T>(page, limit);
        if (orderBy != null)
            queryWrapper.orderBy(true, sort, LetterUtils.valTocamel(orderBy, "_"));
        return baseDao.selectPage(userPage, queryWrapper);
    }

    /**
     * 分页查询
     *
     * @param page         当前页
     * @param limit        每页个数
     * @param orderBy      排序字段
     * @param sort         升序true 降序false
     * @return 查询后的数据
     */
    public IPage<T> selectPage(int page, int limit, String orderBy, boolean sort) {
        QueryWrapper<T> queryWrapper = new QueryWrapper<>();
        IPage<T> userPage = new Page<T>(page, limit);
        if (orderBy != null)
            queryWrapper.orderBy(true, sort, LetterUtils.valTocamel(orderBy, "_"));
        return baseDao.selectPage(userPage, queryWrapper);
    }

    /**
     * 分页查询(baseModel方式1)
     * 实体类继承baseModel后可以直接放入
     *
     * @param baseModel 分页封装的条件
     * @return 查询后的数据
     */
    public IPage<T> selectPage(BaseModel baseModel) {
        return selectPage(baseModel,new QueryWrapper<T>());
    }

    /**
     * 分页查询(baseModel方式2)
     * 实体类继承baseModel后可以直接放入
     * queryWrapper为附加的查询条件
     *
     * @param baseModel    分页的条件
     * @param queryWrapper 封装的条件
     * @return 查询后的数据
     */
    public IPage<T> selectPage(BaseModel baseModel,QueryWrapper<T> queryWrapper) {
        QueryWrapper<T> queryWrapper1 = WrapperUtils.queryWrapperAdd(queryWrapper, baseModel);
        return selectPage(queryWrapper1, baseModel.getPage(), baseModel.getLimit(), baseModel.getOrderBy(), baseModel.getIsSort());
    }

    /**
     * 分页查询(map方式)
     * 实体类继承baseModel后可以直接放入
     * map为附加的查询条件
     *
     * @param baseModel 封装的分页信息
     * @param map       封装的查询条件
     * @return 查询后的数据
     */
    public IPage<T> selectPage(BaseModel baseModel, Map<String, Object> map) {
        QueryWrapper<T> queryWrapper1 = WrapperUtils.mapToQueryWrapper(new QueryWrapper<T>(),map);
        QueryWrapper<T> queryWrapper2 = WrapperUtils.queryWrapperAdd(queryWrapper1, baseModel);
        return selectPage(baseModel,queryWrapper2);
    }
    /**
     * 分页查询(map方式1)
     *
     * @param map       封装的查询条件
     * @param page         当前页
     * @param limit        每页个数
     * @return 查询后的数据
     */
    public IPage<T> selectPage(Map<String, Object> map, int page, int limit) {
        QueryWrapper<T> queryWrapper = WrapperUtils.mapToQueryWrapper(new QueryWrapper<T>(),map);
        return selectPage(queryWrapper,page,limit);
    }

    /**
     * 分页查询(map方式2)
     *
     * @param map       封装的查询条件
     * @param page         当前页
     * @param limit        每页个数
     * @param orderBy      排序字段
     * @param sort         升序true 降序false
     * @return 查询后的数据
     */
    public IPage<T> selectPage(Map<String, Object> map, int page, int limit, String orderBy, boolean sort) {
        QueryWrapper<T> queryWrapper = WrapperUtils.mapToQueryWrapper(new QueryWrapper<T>(),map);
        return selectPage(queryWrapper,page,limit,orderBy,sort);
    }

    /**
     * 分页查询(实体类方式)
     * 实体类继承baseModel后可以直接放入
     * entity为附加的查询条件
     *
     * @param baseModel 封装的分页信息
     * @param entity    封装的查询条件
     * @return 查询后的数据
     */
    @Transactional
    public IPage<T> selectPage(BaseModel baseModel, T entity) {
        QueryWrapper<T> queryWrapper = WrapperUtils.modelToQueryWrapper(entity);
        return selectPage(baseModel,queryWrapper);
    }

    /**
     * 分页查询(实体类方式1)
     *
     * @param entity    封装的查询条件
     * @param page         当前页
     * @param limit        每页个数
     * @return 查询后的数据
     */
    @Transactional
    public IPage<T> selectPage(T entity, int page, int limit) {
        QueryWrapper<T> queryWrapper = WrapperUtils.modelToQueryWrapper(entity);
        return selectPage(queryWrapper,page,limit);
    }

    /**
     * 分页查询(实体类方式2)
     *
     * @param entity    封装的查询条件
     * @param page         当前页
     * @param limit        每页个数
     * @param orderBy      排序字段
     * @param sort         升序true 降序false
     * @return 查询后的数据
     */
    @Transactional
    public IPage<T> selectPage(T entity, int page, int limit, String orderBy, boolean sort) {
        QueryWrapper<T> queryWrapper = WrapperUtils.modelToQueryWrapper(entity);
        return selectPage(queryWrapper,page,limit,orderBy,sort);
    }

    /**
     * 自定义分页封装条件
     *
     * @param page    分页信息
     * @param Wrapper 构造器
     * @return 查询后的数据
     */
    public IPage<T> selectPage(IPage<T> page, Wrapper<T> Wrapper) {
        return baseDao.selectPage(page, Wrapper);
    }

    /**
     * 根据实体类计算数量
     * @param entity 要查询的实体类
     * @return 数量
     */
    public int countByEntity(BaseModel entity){
        QueryWrapper<T> queryWrapper = WrapperUtils.queryWrapperAdd(new QueryWrapper<T>(), entity);
        return baseDao.selectCount(queryWrapper);
    }

    /**
     * 根据实体类计算数量
     * @param entity 要查询的实体类
     * @return 数量
     */
    public int countByEntity(T entity){
        QueryWrapper<T> queryWrapper = WrapperUtils.modelToQueryWrapper(entity);
        return baseDao.selectCount(queryWrapper);
    }

}
